package com.education.service.oss.service.impl;

import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import com.education.service.oss.service.FileService;
import com.education.service.oss.utils.OssProperties;
import org.joda.time.DateTime;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import javax.annotation.PostConstruct;
import java.io.InputStream;
import java.util.UUID;

/**
 * 实现文件上传业务功能
 * Created by Lance on 2020/06/23 17:32
 */
@Service
public class FileServiceImpl implements FileService {

     @Autowired
    private OssProperties ossProperties;
    private String endpoint = null;
    private String keyId = null;
    private String keySecret = null;
    private String bucketName = null;

     @PostConstruct
     public void init(){
         // Endpoint以杭州为例，其它Region请按实际情况填写。
         endpoint = ossProperties.getEndpoint();
         // 阿里云主账号AccessKey拥有所有API的访问权限，风险很高。强烈建议您创建并使用RAM账号进行API访问或日常运维，请登录 https://ram.console.aliyun.com 创建RAM账号。
         keyId = ossProperties.getKeyId();
         keySecret = ossProperties.getKeySecret();
         bucketName = ossProperties.getBucketName();
     }

    @Override
    public String upload(InputStream inputStream, String module, String oFilename) {
        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(endpoint, keyId, keySecret);

        // 接收日期格式 生成文件目录
        String folder = new DateTime().toString("yyyy/MM/dd");
        // 截取 拼接文件扩展名
        String extendName = oFilename.substring(oFilename.lastIndexOf("."));
        // 使用生成的 UUID 做文件名
        String fileName = UUID.randomUUID().toString();

        // 开始拼接文件名
        String key = module + "/" + folder + "/" + fileName + extendName;

        // 上传文件 : 文件路径 avatar/2020/06/23 文件名.扩展名
        ossClient.putObject(bucketName, key, inputStream);

        // 关闭 OSSClient
        ossClient.shutdown();
        // 返回 url :  https://online-education-lance.oss-cn-beijing.aliyuncs.com/avatar/heard.jpg
        // https:// + bucketName + . + endpoint + / + key
        return "https://" + bucketName + "." + endpoint + "/" + key;
    }

    @Override
    public void removeFile(String url) {
        // 拼接主机地址
        String host = "https://" + bucketName + "." + endpoint + "/";
        // 截取 url
        String objectName = url.substring( host.length());

        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(endpoint, keyId, keySecret);

        // 删除文件。如需删除文件夹，请将ObjectName设置为对应的文件夹名称。如果文件夹非空，则需要将文件夹下的所有object删除后才能删除该文件夹。
        ossClient.deleteObject(bucketName, objectName);

        // 关闭OSSClient。
        ossClient.shutdown();
    }
}
